package com.frank.utils;

/**
 * @Description 单向链表
 * @Author zhoujr
 * @Data 2024/11/23 14:35
 */
public class MyLinkedList {

    // 元素个数
    Integer size;

    // 虚拟头节点
    ListNode head;

    public MyLinkedList() {
        this.size = 0;
        this.head = new ListNode(-1);
    }

    /**
     * @Description 获取链表中第 index 个节点的值。如果索引无效，则返回-1
     */
    public int get(int index) {

        //如果index非法，返回-1
        if (index < 0 || index >= size) {
            return -1;
        }
        ListNode temp = head.next;
        for (int i = 0; i < index; i++) {
            temp = temp.next;
        }
        return temp.val;
    }

    /**
     * @Description 在链表的第一个元素之前添加一个值为 val 的节点。插入后，新节点将成为链表的第一个节点。
     */
    public void addAtHead(int val) {
        this.head.next = new ListNode(val, this.head.next);
        this.size++;
    }

    /**
     * @Description 将值为 val 的节点追加到链表的最后一个元素
     */
    public void addAtTail(int val) {
        ListNode temp = this.head;
        while (temp.next != null) {
            temp = temp.next;
        }
        temp.next = new ListNode(val);
        this.size++;
    }

    /**
     * @Description 在链表中的第 index 个节点之前添加值为 val  的节点。如果 index 等于链表的长度，则该节点将附加到链表的末尾。如果 index 大于链表长度，则不会插入节点。如果index小于0，则在头部插入节点。
     */
    public void addAtIndex(int index, int val) {
        if (index > size) {
            return;
        }
        if (index < 0) {
            index = 0;
        }
        ListNode temp = head;
        for (int i = 0; i < index; i++) {
            temp = temp.next;
        }
        temp.next = new ListNode(val, temp.next);
        this.size++;
    }

    /**
     *
     * @Description 如果索引 index 有效，则删除链表中的第 index 个节点
     */
    public void deleteAtIndex(int index) {
        if (index < 0 || index >= size) {
            return;
        }
        ListNode temp = head;
        for (int i = 0; i < index; i++) {
            temp = temp.next;
        }
        temp.next = temp.next.next;
        this.size--;
    }

}

/**
 * Your MyLinkedList object will be instantiated and called as such:
 * MyLinkedList obj = new MyLinkedList();
 * int param_1 = obj.get(index);
 * obj.addAtHead(val);
 * obj.addAtTail(val);
 * obj.addAtIndex(index,val);
 * obj.deleteAtIndex(index);
 */
